package com.project.mq.mqserver.datacenter;


import com.project.mq.MqApplication;
import com.project.mq.mqserver.core.Binding;
import com.project.mq.mqserver.core.Exchange;
import com.project.mq.mqserver.core.ExchangeType;
import com.project.mq.mqserver.core.MSGQueue;
import com.project.mq.mqserver.mapper.MetaMapper;

import java.io.File;
import java.util.List;

/**
 * 通过这个类来封装针对数据库的操作
 */
public class DataBaseManager {
    // 由于 DataBaseManager 不是一个 Bean
    // 需要手动来获取实例
    private MetaMapper metaMapper;
    //一般谈到初始化，会用到构造方法。
    //但是咱们此处，是单独拎出来一个普通的方法。
    //构造方法，一般是用来初始化类的属性~~ 一般不太会涉及到太多的业务逻辑~~
    //此处的初始化，带有业务逻辑的，还是单独拎出来，手动来调用比较合适一些~~
    //数据库初始化=建库建表+插入一些默认数据
    //我们期望，在咱们的broker server启动的时候，做出下列逻辑判定：
    //1. 如果数据库已经存在了（表啥的都有了），不做任何操作。
    //2. 如果数据库不存在，则创建库，创建表，构造默认数据。
    //例如，现在把broker server部署到一个新的服务器上，显然，此时是没有数据库。
    // 就需要让broker server启动的时候，自动的把对应的数据库创建好~~
    // 但是如果是一个已经部署过的机器，broker server重启了，就会发现，数据库已经有了，此时不做任何数据库相关操作。
    //啥叫存在，啥叫不存在？就判定meta.db这个文件是否存在即可!!!
    public void init() {
        //手动获取到MetaMapper
        this.metaMapper = MqApplication.context.getBean(MetaMapper.class);
        //this.metaMapper = JavaMessageQueueApplication.ac.getBean(MetaMapper.class);

        // 构造数据库
        if (!checkDBExists()) {
            //数据库不存在,就进行建库建表操作
            //先创建一个data目录
            File dataDir = new File("./data");
            dataDir.mkdirs();
            // 1. 读取 sql 文件中的内容，并创建表
            createTable();
            // 2. 插入默认数据
            createDefaultData();
            System.out.println("[DataBaseManager] 数据库初始化完成!");
        } else {
            System.out.println("[DataBaseManager] 数据库已经存在!");
        }
    }
    public void deleteDB() {
        File file = new File("./data/meta.db");
        boolean ret = file.delete();
        if (ret) {
            System.out.println("[DataBaseManager] 删除数据库文件成功!");
        } else {
            System.out.println("[DataBaseManager] 删除数据库文件失败!");
        }
        File dataDir = new File("./data");
        //使用 delete 删除目录的时候，需要保证目录是空的.
        ret = dataDir.delete();
        if (ret) {
            System.out.println("[DataBaseManager] 删除数据库目录成功!");
        } else {
            System.out.println("[DataBaseManager] 删除数据库目录失败!");
        }
    }
    private boolean checkDBExists() {
        File file = new File("./data/meta.db");
        if (file.exists()) {
            return true;
        }
        return false;
    }
    // 这个方法用来建表.
    // 建库操作并不需要手动执行.（不需要手动创建meta.db文件）
    // 首次执行这里的数据库操作的时候，就会自动的创建出meta.db文件来（MyBatis帮我们完成的）
    private void createTable() {
        metaMapper.createExchangeTable();
        metaMapper.createQueueTable();
        metaMapper.createBindingTable();
        System.out.println("[DataBaseManager] 创建表完成!");
    }
    // 给数据库表中，添加默认的数据.
    // 此处主要是添加一个默认的交换机.
    // RabbitMQ 里有一个这样的设定：带有一个匿名的交换机，类型是 DIRECT.
    private void createDefaultData() {
        // 构造默认交换机
        Exchange exchange = new Exchange();
        exchange.setName("");
        exchange.setType(ExchangeType.DIRECT);
        exchange.setDurable(true);
        exchange.setAutoDelete(false);
        metaMapper.insertExchange(exchange);
        System.out.println("[DataBaseManager] 创建初始数据完成!");
    }
    //封装其他数据库操作
    public void insertExchange(Exchange exchange) {
        metaMapper.insertExchange(exchange);
    }

    public void deleteExchange(String exchangeName) {
        metaMapper.deleteExchange(exchangeName);
    }

    public List<Exchange> selectAllExchanges() {
        return metaMapper.selectAllExchanges();
    }

    public void insertQueue(MSGQueue queue) {
        metaMapper.insertQueue(queue);
    }

    public void deleteQueue(String queueName) {
        metaMapper.deleteQueue(queueName);
    }

    public List<MSGQueue> selectAllQueues() {
        return metaMapper.selectAllQueues();
    }

    public void insertBinding(Binding binding) {
        metaMapper.insertBinding(binding);
    }

    public void deleteBinding(Binding binding) {
        metaMapper.deleteBinding(binding);
    }

    public List<Binding> selectAllBindings() {
        return metaMapper.selectAllBindings();
    }
}
